from shuangti.Item01.AIAssistant import AIAssistant


class DataAnalysisAssistant(AIAssistant):
    """数据分析助手子类，支持数据格式为自定义属性"""
    def __init__(self, aid: str, name: str, version: str, is_online: bool, ability_score: int, supported_formats: str):
        super().__init__(aid, name, "数据分析助手", version, is_online, ability_score)
        self.supported_formats = supported_formats  # 自定义属性：支持的数据格式（如CSV、Excel）

    def execute_task(self, task_content: str) -> str:
        """执行数据分析任务"""
        if not self.is_online:
            return f"❌ {self.name} 当前离线，无法执行任务。"
        return (f"✅ {self.name}（数据分析助手）已执行任务：\n任务内容：{task_content}\n"
                f"处理结果：已生成数据洞察报告，支持格式：{self.supported_formats}")

    def __str__(self) -> str:
        """补充子类属性的简要信息"""
        base_info = super().__str__()
        return f"{base_info}, 支持数据格式: {self.supported_formats}"